tensor([[0.0300, 0.2900, 0.9700, 0.1000, 0.2100],
        [0.9700, 0.7100, 0.0300, 0.9000, 0.7900]], grad_fn=<RoundBackward1>)